from langchain_openai import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import StrOutputParser
import os
# 定义模型
llm = ChatOpenAI(model_name="deepseek-chat", api_key=os.environ["DEEPSEEK_API_KEY"],base_url=os.environ["DEEPSEEK_BASE_URL"])

# 定义提示模板  专为 聊天模型设计，生成 结构化消息列表，支持多角色（如 system、user、assistant）
prompt_template = ChatPromptTemplate.from_messages([
    ('system','请将以下的内容翻译成{language}'),
    ('user','{text}')
])

parse = StrOutputParser()

chain = prompt_template | llm | parse

for chunk in chain.stream({'text':'我叫你一声，你敢答应吗？','language':'英语'}):
    print(chunk,end='|',flush=True)
